#!/bin/bash

# 这个脚本现在可以接受一个可选的markdown文件路径作为参数
# 如果提供了参数，显示该文件内容；否则显示$RESULT

MARKDOWN_FILE="$1"

# 确定要显示的目录
if [ -n "$MARKDOWN_FILE" ] && [ -f "$MARKDOWN_FILE" ]; then
 # 使用参数指定的文件
 DISPLAY_DIR=$(dirname "$MARKDOWN_FILE")
else
 # 使用临时保存目录
 DISPLAY_DIR="$RFM_TEMP_SAVE_PATH"
fi

rfm -n -d $DISPLAY_DIR &
rfm_PID_DisplayingResult=$!

echo
echo "=== 模型回复 ==="

# 显示内容
if [ -n "$MARKDOWN_FILE" ] && [ -f "$MARKDOWN_FILE" ]; then
 # 显示指定文件的内容
 if which bat >/dev/null 2>&1; then
 bat -p --language=md "$MARKDOWN_FILE"
 else
 cat "$MARKDOWN_FILE"
 fi
else
 # 显示内存中的结果
 if which bat >/dev/null 2>&1; then
 echo "$RESULT" | bat -p --language=md
 else
 echo "$RESULT"
 echo
 echo "安装 bat 软件(gentoo 上的包是sys-apps/bat)可以语法高亮显示当前markdown内容"
 fi
fi

read -p "是否显示原始回复json? [y/N] " displayJSON
displayJSON=${displayJSON:-N}
if [[ "$displayJSON" =~ ^[Yy]$ ]]; then
 echo $RESPONSE|jq
fi
